from django.db import models
from django.contrib.auth import get_user_model

from users.models import BaseModel
from goods.models import Goods

UserProfile = get_user_model()


class ShoppingCart(BaseModel):
    """
    购物车
    """
    user = models.ForeignKey(UserProfile, on_delete=models.CASCADE, verbose_name="用户")
    goods = models.ForeignKey(Goods, on_delete=models.CASCADE, verbose_name=u"商品")
    nums = models.IntegerField(default=0, verbose_name=u"购买数量")

    class Meta:
        verbose_name = '购物车'
        verbose_name_plural = verbose_name
        unique_together = ("user", "goods")

    def __str__(self):
        return "{}({})".format(self.goods.name, self.nums)


class OrderInfo(BaseModel):
    """
    订单
    """
    ORDER_STATUS = (
        ("TRADE_SUCCESS", "成功"),
        ("TRADE_CLOSED", "超时关闭"),
        ("WAIT_BUYER_PAY", "交易创建"),
        ("TRADE_FINISHED", "交易结束"),
        ("paying", "待支付"),
    )
    user = models.ForeignKey(UserProfile, on_delete=models.CASCADE, verbose_name="用户")
    order_sn = models.CharField(max_length=30, null=True, blank=True, unique=True, verbose_name='订单编号')
    trade_no = models.CharField(max_length=100, null=True, blank=True, unique=True, verbose_name="交易编号")
    pay_status = models.CharField(choices=ORDER_STATUS, default="paying", max_length=30, verbose_name=u"订单状态")
    post_script = models.CharField(max_length=200, verbose_name='订单留言')
    order_amount = models.FloatField(default=0.0, verbose_name='订单金额')
    pay_time = models.DateTimeField(auto_now_add=True, verbose_name='支付时间')
    address = models.CharField(max_length=100, default="", verbose_name="收货地址")
    signer_name = models.CharField(max_length=20, default="", verbose_name='签收人')
    siner_mobile = models.CharField(max_length=11, default="", verbose_name="联系电话")

    class Meta:
        verbose_name = '订单'
        verbose_name_plural = verbose_name

    def __str__(self):
        return str(self.order_sn)


class OrderGoods(BaseModel):
    """订单商品详情"""
    order = models.ForeignKey(OrderInfo, on_delete=models.CASCADE, related_name="orders", verbose_name="订单详情")
    goods = models.ForeignKey(Goods, on_delete=models.CASCADE, verbose_name="商品")
    goods_num = models.IntegerField(default=0, verbose_name="商品数量")

    class Meta:
        verbose_name = '订单商品详情'
        verbose_name_plural = verbose_name

    def __str__(self):
        return str(self.order.order_sn)
